from method import *
 
if __name__ == '__main__':
    labels_path = 'C:/myself/hw_ai/ai_datasets/fire/train/labels/'
    delete_folder(labels_path)
    source_path = 'C:/myself/hw_ai/ai_datasets/fire/train/xml/'
    label_path = 'C:/myself/hw_ai/ai_datasets/fire/train/labels/'
    xml_to_yolotxt(source_path, label_path)
    
    labels_path = 'C:/myself/hw_ai/ai_datasets/fire/val/labels/'
    delete_folder(labels_path)
    source_path = 'C:/myself/hw_ai/ai_datasets/fire/val/xml/'
    label_path = 'C:/myself/hw_ai/ai_datasets/fire/val/labels/'
    xml_to_yolotxt(source_path, label_path)